_base_ = ["./voc0712.py", "./schedule_2x.py", "./default_runtime.py"]
# model setting
backbone_pretrained_path = \
    R"G:\models\MMCls\resnet18_batch256_20200708-34ab8f90.pth"
train_cfg = dict(gaussian_iou=0.7)
test_cfg = None
model = dict(type="CenterNet",
             pretrained=backbone_pretrained_path,
             train_cfg=train_cfg,
             test_cfg=test_cfg,
             backbone=dict(type="MMCLSResNet", depth=18, out_indices=(3, )),
             neck=dict(type="DeconvUP",
                       in_channels=512,
                       out_channels_list=[256, 128, 64]),
             bbox_head=dict(type="CenterHead",
                            num_classes=20,
                            in_channels=64,
                            train_cfg=train_cfg,
                            test_cfg=test_cfg,
                            loss_heatmap=dict(type="GaussianFocalLoss",
                                              alpha=2.0,
                                              gamma=4.0,
                                              loss_weight=1),
                            loss_wh_size=dict(type="L1Loss",
                                              reduction="mean",
                                              loss_weight=0.1),
                            loss_ct_off=dict(type="L1Loss",
                                             reduction="mean",
                                             loss_weight=1)))